package com.zzh.filter;

import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import org.apache.commons.lang.StringUtils;
import org.springframework.http.HttpRequest;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

/**
 * @author zihong.zheng
 * @version 1.0
 * @date 2019/2/13
 * @time 20:57
 * @description
 */
@Component
public class TokenFilter extends ZuulFilter {
    /**
     * 过滤器类型
     *
     * @return pre- 请求前执行， post-请求后执行
     */
    @Override
    public String filterType() {
        return "pre";
    }

    /**
     * 请求执行顺序，当一个请求在同一个阶段的时候，存在多个过滤器的时候，多个过滤器执行顺序
     * @return
     */
    @Override
    public int filterOrder() {
        return 0;
    }

    /**
     * 判断过滤器是否生效
     * @return true - 生效， false -不生效
     */
    @Override
    public boolean shouldFilter() {
        return true;
    }

    /**
     * 编写过滤器拦截业务逻辑代码
     * @return
     * @throws ZuulException
     */
    @Override
    public Object run() throws ZuulException {

        //案例：拦截所有的服务接口，判断服务接口上是否有传递userToken参数

        //1.获取上下文
        RequestContext currentContext = RequestContext.getCurrentContext();

        //2.获取request
        HttpServletRequest request = currentContext.getRequest();


        //3.获取userToken,从请求参数中获取（一般要求从请求头中获取）
        String userToken = request.getHeader("userToken");
        if(StringUtils.isBlank(userToken)){
            //不会继续执行，不会调用服务接口，服务网关直接响应给客户端
            currentContext.setSendZuulResponse(false);
            currentContext.setResponseBody("userToken is null");
            currentContext.setResponseStatusCode(401);
            return null;
            //返回一个错误提示
        }
        return null;
    }
}
